-
Notifications
You must be signed in to change notification settings - Fork 70
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
adding mask_indices routine #426
Conversation
@ipdemes please please don't force-push changes while the changes are under a review. doing so makes it extremely difficult to look back the history. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Some small documentation fixes, and a complaint about the warning, otherwise looks good.
cunumeric/module.py
Outdated
Notes | ||
----- | ||
WARNING: `mask_indices` expects `mask_function` to call cuNumeric functions | ||
for good performance. In case non-cuNumeric fucntions are called by |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
typo: "functions"
cunumeric/module.py
Outdated
Assume `mask_func` is a function that, for a square array a of size | ||
``(n, n)`` with a possible offset argument `k`, when called as | ||
``mask_func(a, k)`` returns a new array with zeros in certain locations | ||
(functions like `triu` or `tril` do precisely this). Then this function |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
functions like :func:`cunumeric.triu` or :func:`cunumeric.tril`
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
See 2 unresolved comments for some remaining minor documentation fixes. Feel free to merge after that.
Edit: Actually, let's way for mypy to run in CI
@manopapad : all the tests pass after your commit. Can I merge this PR? |
yes, please go ahead |
adding mask_indices routine
adding mask_indices routine
adding mask_indices routine
@magnatelee : is this implementation align with your suggestion?